import netron

'''
view the model by netron

netron have 2 version
1、 netron.exe
2、 import netron 
    netron.start(model_path)
'''
netron.start("modelpath")

from torchvision.models import resnet18

model = resnet18().to("cuda")

'''
torchsummary torchstat two ways to check the detail of the model
'''

from torchsummary import summary

summary(model, (3, 224, 224))

from torchstat import stat

stat(model, (3, 224, 224))
